// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/public/common/mhtml_generation_params.h"

#include "base/command_line.h"
#include "base/files/file_path.h"
#include "content/public/common/content_switches.h"

namespace content {

MHTMLGenerationParams::MHTMLGenerationParams(const base::FilePath& file_path)
    : file_path(file_path)
{
    // Check which variant of MHTML generation is required.
    std::string mhtmlGeneratorOptionFlag = base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
        switches::kMHTMLGeneratorOption);
    if (mhtmlGeneratorOptionFlag == switches::kMHTMLSkipNostoreMain) {
        cache_control_policy = blink::WebFrameSerializerCacheControlPolicy::
            FailForNoStoreMainFrame;
    } else if (mhtmlGeneratorOptionFlag == switches::kMHTMLSkipNostoreAll) {
        cache_control_policy = blink::WebFrameSerializerCacheControlPolicy::
            SkipAnyFrameOrResourceMarkedNoStore;
    }
}

} // namespace content
